[Day9] 簡單搞懂GraphQl API Python 實作範例

2022 iThome 鐵人賽

DAY 9

Software Development

剛入職軟體工程師會需要知道的常見工具篇系列第 9 篇

14th鐵人賽 graphql python fastapi

hello02923

2022-09-23 12:03:27

2979 瀏覽

分享至

昨天簡單介紹後，今天又要來實作啦～

??提醒，如果是利用Homebrew or Linuxbrew 裝 Python的話 pip install 'strawberry-graphql'會有問題！如下圖

DEPRECATION: Configuring installation scheme with distutils config files is deprecated and will no longer work in the near future. If you are using a Homebrew or Linuxbrew Python, please see discussion at https://github.com/Homebrew/homebrew-core/issues/76621

⭐️解決方法：
step1. python3 -m venv {環境名稱}
step2. source ./{環境名稱}/bin/activate
step3. pip strawberry-graphql[debug-server] or strawberry-graphql[fastapi]

好啦！不囉嗦要開始實作囉～

資料夾

└──demo
    └──app.py
    └──data.py
    └──main.py
    └──model.py
    └──demo_graph      ##這邊虛擬環境的一包就不列出來 demo_graph是環境名稱

main.py 主執行檔
- GraphQLRouter是strawberry 支援fastapi的Router，純用fastapi也有原生的APIRouter
- include_router 是協助加入各個router到主要的FastAPI中，所以可以有多個喔

# -*- coding: utf-8 -*-

from fastapi import FastAPI
from strawberry.fastapi import GraphQLRouter
from app import schema

graphql_app = GraphQLRouter(schema)

app = FastAPI()
app.include_router(graphql_app, prefix="/graphql")

model.py 設定資料結構

# -*- coding: utf-8 -*-
import typing
import strawberry
from datetime import datetime

# 用戶資料
@strawberry.type
class User:
    UserID: str
    IsLogin: bool
    Created_at: datetime

# 飲料資訊
@strawberry.type
class Drink:
    channel: str
    name: str
    info: str

來對照API文件頁面~
用戶資料頁面
飲料資料頁面
app.py query資料的架構

# -*- coding: utf-8 -*-
import strawberry
import typing
from data import (
    get_users, get_drinks
)
from model import (
    User, Drink
)

@strawberry.type
class Query:
    users: User = strawberry.field(resolver=get_users)
    drinks: typing.List[Drink] = strawberry.field(resolver=get_drinks)

schema = strawberry.Schema(Query)

data.py 去跟db互動要資料整理..

# -*- coding: utf-8 -*-
"""
取得資料
"""
from datetime import datetime
from pymongo import MongoClient
from model import (
    User, Drink
)
#%%
client = MongoClient('localhost:27017', username='帳號', password='密碼')
#%%
# 取得用戶資料
def get_users():
    return User(
        UserID='Mary',
        IsLogin=False,
        Created_at=datetime.now()
    )

# 取得飲料資料
def get_drinks():
    data = list(
        client['Shop']['Drink'].find(
            {}, {'_id':0, 'channel':1, 'name':1, 'info':1}
        ).limit(3)
    )
    
    return list(Drink(
        channel = item.get('channel'),
        name = item.get('name'),
        info = item.get('info'),
    ) for item in data)